Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Rewrite benchmarking scripts with Python. #5082

Merged
merged 16 commits into from
Mar 16, 2021
Merged

Conversation

hanhanW
Copy link
Contributor

@hanhanW hanhanW commented Mar 12, 2021

This moves the config to utils.py, which makes the compilation script and benchmarking script share the same config. A ModelBenchmarkInfo contains a phone list. This allows us to set different config for different models. This is useful for doing batch mode benchmarking.

@google-cla google-cla bot added the cla: yes label Mar 12, 2021
build_tools/mako/benchmark_modules_on_android.py Outdated Show resolved Hide resolved
build_tools/mako/benchmark_modules_on_android.py Outdated Show resolved Hide resolved
build_tools/mako/benchmark_modules_on_android.py Outdated Show resolved Hide resolved
build_tools/mako/benchmark_modules_on_android.py Outdated Show resolved Hide resolved
build_tools/mako/benchmark_modules_on_android.py Outdated Show resolved Hide resolved
build_tools/mako/utils.py Outdated Show resolved Hide resolved
@hanhanW hanhanW requested a review from GMNGeoffrey March 15, 2021 09:21
build_tools/mako/benchmark_modules_on_android.py Outdated Show resolved Hide resolved

import configuration

PATTERN = re.compile(r"BM_(\w+)(.+)/real_time(\s+) (?P<ms>[.0-9]+) ms")
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I still think we should be using the more structured output instead of parsing the human-readable output... but this is ok for now

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes, I think we can let it report a csv or something. I keep this because this is the same behavior as before. I will consider this.

@hanhanW hanhanW merged commit 5b6e7d1 into iree-org:main Mar 16, 2021
@hanhanW hanhanW deleted the bench-py branch March 16, 2021 05:23
@not-jenni not-jenni mentioned this pull request Mar 16, 2021
copybara-service bot pushed a commit that referenced this pull request Mar 16, 2021
* 9ae2e49 Rename compiler2 to compiler (#5113)
* 4168916 [CodeGen] Add a pattern to canonicalize HAL interface load/store (#5103)
* 5b6e7d1 Rewrite benchmarking scripts with Python. (#5082)
* 36912bc Merge google -> main (#5117)
* ec9f244 Merge branch 'main' into google-to-main
* e479591 Disable native-py check from buildifier action
* 02bb838 Enable concat, gather and torch_index_select on Linalg on tensors path. (#5053)
* 31c71df Integrate MLIR-EmitC at iml130/mlir-emitc@3483f16 (#5114)
* 2839b02 Fix missing <cstddef> header file. (#5112)
* 5670121 Fixing flow.dispatch.workgroups tied operands when removing args. (#5108)

COPYBARA_INTEGRATE_REVIEW=#5125 from not-jenni:main-to-google 4936e55
PiperOrigin-RevId: 363244672
iree-github-actions-bot pushed a commit that referenced this pull request Mar 16, 2021
* 9ae2e49 Rename compiler2 to compiler (#5113)
* 4168916 [CodeGen] Add a pattern to canonicalize HAL interface load/store (#5103)
* 5b6e7d1 Rewrite benchmarking scripts with Python. (#5082)
* 36912bc Merge google -> main (#5117)
* ec9f244 Merge branch 'main' into google-to-main
* e479591 Disable native-py check from buildifier action
* 02bb838 Enable concat, gather and torch_index_select on Linalg on tensors path. (#5053)
* 31c71df Integrate MLIR-EmitC at iml130/mlir-emitc@3483f16 (#5114)
* 2839b02 Fix missing <cstddef> header file. (#5112)
* 5670121 Fixing flow.dispatch.workgroups tied operands when removing args. (#5108)

PiperOrigin-RevId: 363244672
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants